import time
from datetime import datetime

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager

""" 简单截图
# 1. 打开浏览器驱动管理
ChromeIns = ChromeDriverManager().install()
# 2. 创建谷歌浏览器对象
driver = webdriver.Chrome(service=Service(ChromeIns))
# 3. 输入百度网址
driver.get('https://www.baidu.com')
time.sleep(2)
# 4.截图
driver.save_screenshot('image.png') # 图片默认保存在当前路径下
# 5. 关闭浏览器
driver.quit()
"""

""" 进阶版截图，获取截图的当前时间，保存到指定的文件夹中
# 1. 打开浏览器驱动管理
ChromeIns = ChromeDriverManager().install()
# 2. 创建谷歌浏览器对象
driver = webdriver.Chrome(service=Service(ChromeIns))
# 3. 输入百度网址
driver.get('https://www.baidu.com')
time.sleep(2)
# 4.进阶截图
filename = 'test-' + datetime.now().strftime('%Y-%m-%d-%H%M%S') + '.png'
driver.save_screenshot('./images/' + filename) # 图片保存在当前路径下的images文件中
# 5. 关闭浏览器
driver.quit()
"""















